package com.pingan.haofang.searchcloud.common.rpc;

import java.util.HashMap;
import java.util.Map;

import com.pingan.haofang.searchcloud.common.exception.SystemException;

/**
 * RPC工厂
 * 
 * @author LUYI374
 * @date 2017年5月11日
 * @since 1.0.0
 */
public class RPCFactory {

    private static final Map<String, RPCInvoker> FACTORY = new HashMap<String, RPCInvoker>();

    /**
     * 注册执行器
     * 
     * @param beanName
     * @param method
     * @param invoker
     */
    public synchronized static void register(String beanName, String method, RPCInvoker invoker) {
        String key = getKey(beanName, method);
        if (FACTORY.containsKey(key)) {
            throw new SystemException("duplicate rpc definied, name:" + beanName + "." + method);
        }
        FACTORY.put(key, invoker);
    }

    /**
     * 得到执行器
     * 
     * @param beanName
     * @param method
     * @return
     */
    public static RPCInvoker getInvoker(String beanName, String method) {
        String key = getKey(beanName, method);
        return FACTORY.get(key);
    }

    /**
     * 得到执行器
     * 
     * @param key
     * @return
     */
    public static RPCInvoker getInvoker(String key) {
        return FACTORY.get(key);
    }

    /**
     * 获取KEY
     * 
     * @param beanName
     * @param method
     * @return
     */
    private static String getKey(String beanName, String method) {
        return beanName + "." + method;
    }
}
